#define _USE_MATH_DEFINES
#include "../matplotlibcpp.h"
#include <cmath>

namespace plt = matplotlibcpp;

using namespace matplotlibcpp;
using namespace std;

int main() {
  // Prepare data.
  int n = 5000;
  std::vector<double> x(n), y(n), z(n), w(n, 2);
  for (int i = 0; i < n; ++i) {
    x.at(i) = i * i;
    y.at(i) = sin(2 * M_PI * i / 360.0);
    z.at(i) = log(i);
  }

  // Plot line from given x and y data. Color is selected automatically.
  plt::subplot(2, 2, 1);
  plt::plot(x, y);

  // Plot a red dashed line from given x and y data.
  plt::subplot(2, 2, 2);
  plt::plot(x, w, "r--");

  // Plot a line whose name will show up as "log(x)" in the legend.
  plt::subplot(2, 2, 3);
  plt::plot(x, z, {{"label", "log(x)"}});

  // Set x-axis to interval [0,1000000]
  plt::xlim(0, 1000 * 1000);

  // Add graph title
  plt::title("Sample figure");
  // Enable legend.
  plt::legend();

  plt::show(false);

  cout << "matplotlibcpp::show() is working in an non-blocking mode" << endl;
  getchar();
}
